#!/bin/bash

# 设置颜色输出
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

# 创建驱动目录
DRIVERS_DIR="drivers"
mkdir -p $DRIVERS_DIR
cd $DRIVERS_DIR

echo -e "${YELLOW}开始下载 Ignite 驱动文件...${NC}"

# 下载驱动文件
download_file() {
    local url=$1
    local filename=$2
    echo -e "${YELLOW}下载 $filename...${NC}"
    if curl -L -o $filename $url; then
        echo -e "${GREEN}成功下载 $filename${NC}"
    else
        echo -e "${RED}下载 $filename 失败${NC}"
        exit 1
    fi
}

# 下载必要的 JAR 文件
download_file "https://repo1.maven.org/maven2/org/apache/ignite/ignite-core/2.15.0/ignite-core-2.15.0.jar" "ignite-core-2.15.0.jar"
download_file "https://repo1.maven.org/maven2/org/apache/ignite/ignite-indexing/2.15.0/ignite-indexing-2.15.0.jar" "ignite-indexing-2.15.0.jar"
download_file "https://repo1.maven.org/maven2/org/apache/ignite/ignite-spring/2.15.0/ignite-spring-2.15.0.jar" "ignite-spring-2.15.0.jar"
download_file "https://repo1.maven.org/maven2/org/apache/ignite/ignite-log4j/2.15.0/ignite-log4j-2.15.0.jar" "ignite-log4j-2.15.0.jar"

# 验证下载的文件
echo -e "${YELLOW}验证下载的文件...${NC}"
ls -l *.jar

# 创建 DBeaver 驱动配置说明
echo -e "${YELLOW}创建 DBeaver 驱动配置说明...${NC}"
cat > README.md << 'EOF'
# Apache Ignite DBeaver 驱动配置说明

## 驱动文件位置
当前目录下的所有 .jar 文件

## 配置步骤
1. 打开 DBeaver
2. 点击菜单 Database -> Driver Manager
3. 找到 Apache Ignite 驱动
4. 点击 Edit 按钮
5. 在 Libraries 标签页中，点击 Add File
6. 添加以下 JAR 文件：
   - ignite-core-2.15.0.jar
   - ignite-indexing-2.15.0.jar
   - ignite-spring-2.15.0.jar
   - ignite-log4j-2.15.0.jar

## 连接参数
- 驱动类：org.apache.ignite.IgniteJdbcThinDriver
- URL：jdbc:ignite:thin://localhost:10800
- 用户名：ignite
- 密码：ignite123

## 验证连接
1. 点击 Test Connection 按钮
2. 如果配置正确，应该能够成功连接

## 常见问题
1. 如果提示找不到驱动类，请检查 JAR 文件是否正确添加
2. 如果连接失败，请检查：
   - Ignite 服务是否正常运行
   - 端口是否正确开放
   - 认证信息是否正确
EOF

echo -e "${GREEN}驱动文件下载完成！${NC}"
echo -e "${GREEN}请按照 README.md 中的说明在 DBeaver 中配置驱动。${NC}" 